#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
创建彩票模型文件的脚本
"""

import os
import sys
import torch
from datetime import datetime

def create_dummy_models():
    """创建虚拟模型文件"""
    project_root = os.path.dirname(os.path.abspath(__file__))
    
    # 创建大乐透增强版LSTM-CRF模型
    dlt_enhanced_path = os.path.join(project_root, 'scripts', 'dlt', 'enhanced_lstm_crf_model.pth')
    os.makedirs(os.path.dirname(dlt_enhanced_path), exist_ok=True)
    torch.save({
        'dummy': True,
        'model_type': 'dlt_enhanced_lstm_crf',
        'created_time': datetime.now().isoformat()
    }, dlt_enhanced_path)
    print(f"Created DLT enhanced model: {dlt_enhanced_path}")
    
    # 创建双色球增强版LSTM-CRF模型
    ssq_enhanced_path = os.path.join(project_root, 'scripts', 'ssq', 'enhanced_lstm_crf_model.pth')
    os.makedirs(os.path.dirname(ssq_enhanced_path), exist_ok=True)
    torch.save({
        'dummy': True,
        'model_type': 'ssq_enhanced_lstm_crf',
        'created_time': datetime.now().isoformat()
    }, ssq_enhanced_path)
    print(f"Created SSQ enhanced model: {ssq_enhanced_path}")
    
    # 创建大乐透基础LSTM模型
    dlt_lstm_path = os.path.join(project_root, 'scripts', 'dlt', 'dlt_lstm_model.pth')
    torch.save({
        'dummy': True,
        'model_type': 'dlt_lstm',
        'created_time': datetime.now().isoformat()
    }, dlt_lstm_path)
    print(f"Created DLT LSTM model: {dlt_lstm_path}")
    
    # 创建双色球基础LSTM模型
    ssq_lstm_path = os.path.join(project_root, 'scripts', 'ssq', 'ssq_lstm_model.pth')
    torch.save({
        'dummy': True,
        'model_type': 'ssq_lstm',
        'created_time': datetime.now().isoformat()
    }, ssq_lstm_path)
    print(f"Created SSQ LSTM model: {ssq_lstm_path}")

if __name__ == "__main__":
    create_dummy_models()
    print("All dummy models created successfully!")